// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/files/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "net/disk_cache/cache_util.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/platform_test.h"

namespace disk_cache {

class CacheUtilTest : public PlatformTest {
public:
    void SetUp() override
    {
        PlatformTest::SetUp();
        ASSERT_TRUE(tmp_dir_.CreateUniqueTempDir());
        cache_dir_ = tmp_dir_.path().Append(FILE_PATH_LITERAL("Cache"));
        file1_ = base::FilePath(cache_dir_.Append(FILE_PATH_LITERAL("file01")));
        file2_ = base::FilePath(cache_dir_.Append(FILE_PATH_LITERAL(".file02")));
        dir1_ = base::FilePath(cache_dir_.Append(FILE_PATH_LITERAL("dir01")));
        file3_ = base::FilePath(dir1_.Append(FILE_PATH_LITERAL("file03")));
        ASSERT_TRUE(base::CreateDirectory(cache_dir_));
        FILE* fp = base::OpenFile(file1_, "w");
        ASSERT_TRUE(fp != NULL);
        base::CloseFile(fp);
        fp = base::OpenFile(file2_, "w");
        ASSERT_TRUE(fp != NULL);
        base::CloseFile(fp);
        ASSERT_TRUE(base::CreateDirectory(dir1_));
        fp = base::OpenFile(file3_, "w");
        ASSERT_TRUE(fp != NULL);
        base::CloseFile(fp);
        dest_dir_ = tmp_dir_.path().Append(FILE_PATH_LITERAL("old_Cache_001"));
        dest_file1_ = base::FilePath(dest_dir_.Append(FILE_PATH_LITERAL("file01")));
        dest_file2_ = base::FilePath(dest_dir_.Append(FILE_PATH_LITERAL(".file02")));
        dest_dir1_ = base::FilePath(dest_dir_.Append(FILE_PATH_LITERAL("dir01")));
    }

protected:
    base::ScopedTempDir tmp_dir_;
    base::FilePath cache_dir_;
    base::FilePath file1_;
    base::FilePath file2_;
    base::FilePath dir1_;
    base::FilePath file3_;
    base::FilePath dest_dir_;
    base::FilePath dest_file1_;
    base::FilePath dest_file2_;
    base::FilePath dest_dir1_;
};

TEST_F(CacheUtilTest, MoveCache)
{
    EXPECT_TRUE(disk_cache::MoveCache(cache_dir_, dest_dir_));
    EXPECT_TRUE(base::PathExists(dest_dir_));
    EXPECT_TRUE(base::PathExists(dest_file1_));
    EXPECT_TRUE(base::PathExists(dest_file2_));
    EXPECT_TRUE(base::PathExists(dest_dir1_));
#if defined(OS_CHROMEOS)
    EXPECT_TRUE(base::PathExists(cache_dir_)); // old cache dir stays
#else
    EXPECT_FALSE(base::PathExists(cache_dir_)); // old cache is gone
#endif
    EXPECT_FALSE(base::PathExists(file1_));
    EXPECT_FALSE(base::PathExists(file2_));
    EXPECT_FALSE(base::PathExists(dir1_));
}

TEST_F(CacheUtilTest, DeleteCache)
{
    disk_cache::DeleteCache(cache_dir_, false);
    EXPECT_TRUE(base::PathExists(cache_dir_)); // cache dir stays
    EXPECT_FALSE(base::PathExists(dir1_));
    EXPECT_FALSE(base::PathExists(file1_));
    EXPECT_FALSE(base::PathExists(file2_));
    EXPECT_FALSE(base::PathExists(file3_));
}

TEST_F(CacheUtilTest, DeleteCacheAndDir)
{
    disk_cache::DeleteCache(cache_dir_, true);
    EXPECT_FALSE(base::PathExists(cache_dir_)); // cache dir is gone
    EXPECT_FALSE(base::PathExists(dir1_));
    EXPECT_FALSE(base::PathExists(file1_));
    EXPECT_FALSE(base::PathExists(file2_));
    EXPECT_FALSE(base::PathExists(file3_));
}

TEST_F(CacheUtilTest, DeleteCacheFile)
{
    EXPECT_TRUE(disk_cache::DeleteCacheFile(file1_));
    EXPECT_FALSE(base::PathExists(file1_));
    EXPECT_TRUE(base::PathExists(cache_dir_)); // cache dir stays
    EXPECT_TRUE(base::PathExists(dir1_));
    EXPECT_TRUE(base::PathExists(file3_));
}

} // namespace disk_cache
